/* bzflag
 * Copyright (c) 1993-2020 Tim Riker
 *
 * This package is free software;  you can redistribute it and/or
 * modify it under the terms of the license found in the file
 * named COPYING that should have accompanied this file.
 *
 * THIS PACKAGE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 */

/* interface header */
#include "DirectoryNames.h"

/* implementation system headers */
#ifndef _WIN32
#  include <stdlib.h>
#  include <unistd.h>
#  include <sys/types.h>
#  include <pwd.h>
#else   // _WIN32
#  include <sys/types.h>
#  include <sys/stat.h>
#  include <stdio.h>
#  include <direct.h>
#  include <shlobj.h>
#endif  // _WIN32
#if defined(__APPLE__)
#  include <CoreServices/CoreServices.h>
#endif


// NOTE: terminate all strings with '/' or '\\'

std::string configDir(bool set, const char *str)
{
    static std::string customConfigDir = std::string("");
    if (set)
        customConfigDir = std::string(str);
    return customConfigDir;
}


void            setCustomConfigDir(const char *str)
{
    std::string dir = str;
    if (dir.size() > 2)
    {
        if (*dir.begin() == '"')
            dir.erase(dir.begin());

        if (*(dir.end()-1) == '"')
            dir.erase(dir.end()-1);

    }
    configDir(1, dir.c_str());
}


std::string     getConfigDirName( const char* versionName )
{
    std::string customConfigDir = configDir(0, NULL);

    if (customConfigDir.size() > 0)
    {
        std::string customName = customConfigDir + DirectorySeparator;
        if (versionName)
        {
            customName += versionName;
            customName += DirectorySeparator;
        }
        return customName;
    }

#if defined(_WIN32)
    std::string name("C:");
    char dir[MAX_PATH];
    ITEMIDLIST* idl;
    if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, CSIDL_LOCAL_APPDATA, &idl)))
    {
        if (SHGetPathFromIDList(idl, dir))
        {
            struct stat statbuf;
            if (stat(dir, &statbuf) == 0 && (statbuf.st_mode & _S_IFDIR) != 0)
                name = dir;
        }

        IMalloc* shalloc;
        if (SUCCEEDED(SHGetMalloc(&shalloc)))
        {
            shalloc->Free(idl);
            shalloc->Release();
        }
    }

    name += "\\BZFlag\\";
    if (versionName)
    {
        name += versionName;
        name += "\\";
    }
    return name;

#elif defined(__APPLE__)
    std::string name;
    ::FSRef libraryFolder;
    ::OSErr err;
    err = ::FSFindFolder(::kUserDomain, ::kApplicationSupportFolderType, true, &libraryFolder);
    if (err == ::noErr)
    {
        char buff[1024];
        err = ::FSRefMakePath(&libraryFolder, (UInt8*)buff, sizeof(buff));
        if (err == ::noErr)
        {
            name = buff;
            name += "/BZFlag/";
            if (versionName)
            {
                name += versionName;
                name += "/";
            }
        }
    }
    return name;
#else
    std::string name;
    struct passwd *pwent = getpwuid(getuid());
    if (pwent && pwent->pw_dir)
    {
        name += std::string(pwent->pw_dir);
        name += "/";
    }
    name += ".bzf/";
    if (versionName)
    {
        name += versionName;
        name += "/";
    }
    return name;
#endif
}

static std::string      setupString(std::string dir)
{
    std::string name = getConfigDirName();
    name += dir;
    name += DirectorySeparator;
    return name;
}

std::string getCacheDirName()
{
    std::string name = getConfigDirName() + "cache" + DirectorySeparator;
    return name;
}


std::string getRecordDirName()
{
    return setupString("recordings");
}

std::string getScreenShotDirName()
{
    return setupString("screenshots");
}

std::string getTempDirName()
{
// FIXME: needs something for Windows and maybe other platforms
#if defined(_WIN32)
    std::string name = getConfigDirName();
    name += "temp";
#else
    std::string name;
    if (getenv("TMPDIR"))
        name = getenv("TMPDIR");
    else
        name = "/tmp";
#endif
    name += DirectorySeparator;
    return name;
}

std::string getWorldDirName()
{
    return setupString("worlds");
}

// Local Variables: ***
// mode: C++ ***
// tab-width: 4 ***
// c-basic-offset: 4 ***
// indent-tabs-mode: nil ***
// End: ***
// ex: shiftwidth=4 tabstop=4
